"""
自定义装饰器，验证某些请求是否登陆过了，如果已经登录过了，则继续进行其他操作，否则进行拦截
"""
from flask import g


def login_required(func):
    def wrappes(*args, **kwargs):
        if g.user_id is not None:  # id不为空，就代表登录过
            # 如果登录以后，则可以继续进行下面的操作
            return func(*args, **kwargs)
        else:
            return {'message': '未登录， 请先进行登录'}

    return wrappes
